package com.example.lettcode._202410._20241030;

import com.example.lettcode.node.ListNode;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/*
25. K 个一组翻转链表
给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。
k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

示例 1：
输入：head = [1,2,3,4,5], k = 2
输出：[2,1,4,3,5]

示例 2：
输入：head = [1,2,3,4,5], k = 3
输出：[3,2,1,4,5]
 */
public class _1_K_ge_yi_zu_fan_zhuan_lian_biao {

    public static void main(String[] args) {
        ListNode list1 = new ListNode(1);
        ListNode list2 = new ListNode(2);
        ListNode list3 = new ListNode(3);
        ListNode list4 = new ListNode(4);
        ListNode list5 = new ListNode(5);
        list1.next = list2;
        list2.next = list3;
        list3.next = list4;
        list4.next = list5;
        ListNode listNode = reverseKGroup2(list1, 2);
        System.out.println(listNode);
    }


    /*
        分段区间反转
     */
    public static ListNode reverseKGroup2(ListNode head, int k) {
        if (head == null || head.next == null || k == 0) {
            return head;
        }

        ListNode node = new ListNode(-1, head);
        ListNode pre = node;
        ListNode end = node;

        while (end != null) {
            for (int i = 0; i < k && end != null; i++) {
                end = end.next;
            }

            if (end == null) {
                break;
            }

            ListNode start = pre.next;
            ListNode nextStart = end.next;
            end.next = null;
            pre.next = reverse(start);
            start.next = nextStart;

            pre = start;
            end = start;
        }
        return node.next;
    }

    private static ListNode reverse(ListNode node) {
        ListNode prev = null;
        ListNode cur = node;

        while(cur != null) {
            ListNode temp = cur.next;
            cur.next = prev;
            prev = cur;
            cur = temp;
        }
        return prev;
    }

    /*
        暴力破解
     */
    public static ListNode reverseKGroup(ListNode head, int k) {
        if (head == null || head.next == null || k == 0) {
            return head;
        }

        Map<Integer, List<ListNode>> map = new HashMap<>();

        ListNode node = head;

        List<ListNode> list = new ArrayList<>();
        while (node != null) {
            list.add(node);
            node = node.next;
        }

        int count = 0;
        int oldSize = list.size() / k;
        int size = list.size() % k == 0 ? list.size() / k : list.size() / k + 1;
        int tempK = k;

        for (int i = 0; i < size; i++) {
            List<ListNode> mapList = new ArrayList<>();
            if (map.get(i) == null) {
                map.put(i, mapList);
            } else {
                mapList = map.get(i);
            }
            if (i == size - 1 && size != oldSize) {
                // 最后的剩余的数量不足K个
                tempK = list.size() - (size - 1) * k;
            }

            for (int j = 0; j < tempK; j++) {
                mapList.add(list.get(count));
                count++;
            }
        }

        List<ListNode> listNodes = map.get(size - 1);
        ListNode listNode1 = listNodes.get(0);
        listNode1.next = null;

        ListNode node2 = new ListNode(-1);
        ListNode node3 = node2;
        for (Map.Entry<Integer, List<ListNode>> entry : map.entrySet()) {
            List<ListNode> value = entry.getValue();
            if (value.size() == k) {
                for (int i = value.size() - 1; i >= 0 ; i--) {
                    ListNode listNode = value.get(i);
                    node3.next = listNode;
                    node3 = node3.next;
                }
            } else {
                for (int i = 0; i < value.size(); i++) {
                    ListNode listNode = value.get(i);
                    node3.next = listNode;
                    node3 = node3.next;
                }
            }

        }

        return node2.next;
    }
}
